package com.dh.leetcode.k1;

import org.junit.Assert;
import org.junit.Test;

/**
 * ClassName: _1137_n_th_tribonacci_number
 *
 * @author shouzimu
 * @Description: 1137. 第N个泰波那契数
 * 泰波那契序列 Tn 定义如下：
 * <p>
 * T0 = 0, T1 = 1, T2 = 1, 且在 n >= 0 的条件下 Tn+3 = Tn + Tn+1 + Tn+2
 * <p>
 * 给你整数 n，请返回第 n 个泰波那契数 Tn 的值。
 * <p>
 * <p>
 * <p>
 * 示例 1：
 * <p>
 * 输入：n = 4
 * 输出：4
 * 解释：
 * T_3 = 0 + 1 + 1 = 2
 * T_4 = 1 + 1 + 2 = 4
 * 示例 2：
 * <p>
 * 输入：n = 25
 * 输出：1389537
 * <p>
 * <p>
 * https://leetcode-cn.com/problems/n-th-tribonacci-number/
 * @date 2021/8/8 20:39
 */
public class _1137_n_th_tribonacci_number {
    public int tribonacci(int n) {
        int f0 = 0;
        int f1 = 1;
        int f2 = 1;
        if (n == 0) {
            return f0;
        } else if (n == 1 || n == 2) {
            return 1;
        }
        int tmp;
        for (int i = 3; i <= n; i++) {
            tmp = f2;
            f2 = f2 + f0 + f1;
            f0 = f1;
            f1 = tmp;
        }
        return f2;
    }

    @Test
    public void tribonacciTest() {
        Assert.assertEquals(2, tribonacci(3));
        Assert.assertEquals(4, tribonacci(4));
        Assert.assertEquals(1389537, tribonacci(25));
    }
}
